#!user/bin/env python3
# -- coding: utf-8 --
# @Createtime：2023/9/7 14:25
# @Updatetime：14:25
# @Author：Test008
# @File：1
# @Description：

from pyecharts.charts import *
from pyecharts.globals import ThemeType
from pyecharts.options import *
from pyecharts import options

# 1》打开数据，处理数据
# 2》
# 读取数据
with open("../../../static/doc/yiqing_data/1960-2019全球GDP数据.csv", "r", encoding="GB2312") as f:
    data_lines = f.readlines()
# 删除第一条数据
data_lines.pop(0)
# 将数据装欢为字典存储，格式为：
# {年份：[[国家，GDP], [国家，GDP]......], [[国家，GDP], [国家，GDP]......],......}
# 先定义一个空字典
data_dict = {}
for line in data_lines:
    year = int(line.split(",")[0])  # year
    country = line.split(",")[1]  # country
    gdp = float(line.split(",")[2])  # gdp

    # 如何判断字典里面有没有指定的key呢？
    try:
        data_dict[year].append([country, gdp])
    except KeyError:
        data_dict[year] = []
        data_dict[year].append([country, gdp])

timeline = Timeline(
    init_opts=options.InitOpts(theme=ThemeType.WONDERLAND)
)

sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    data_dict[year].sort(key=lambda ele: ele[1], reverse=True)
    # 取出本年度前10名的国家
    year_data = data_dict[year][0:20]
    # 添加x，y轴数据
    x_data = []
    y_data = []
    for country_gdp in year_data:
        x_data.append(country_gdp[0])  # x添加国家
        y_data.append(country_gdp[1] / 100000000)  # y轴添加gdp

    # 构建柱状图
    bar = Bar()
    x_data.reverse()
    y_data.reverse()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP(亿)", y_data, label_opts=LabelOpts(position='right'))
    # 反转x轴，y轴
    bar.reversal_axis()
    # 设置每一年的标题
    bar.set_global_opts(
        title_opts=TitleOpts(title=f"{year}全球前20 GDP数据")
    )
    # 添加时间线
    timeline.add(bar, str(year))

# 设置时间线自动播放
timeline.add_schema(
    play_interval=500,
    is_timeline_show=True,
    is_auto_play=True,
    is_loop_play=True
)

# 绘图
timeline.render()
